package com.zzt.questionbankusermanager.bean.tools;

import com.zzt.questionbankusermanager.bean.dto.MenuItemDTO;
import com.zzt.questionbankusermanager.bean.dto.MenuList;

import java.util.ArrayList;
import java.util.List;

public class MenuCreater {
    public MenuList createMenu(Long id , List< MenuItemDTO > dtos){
        MenuList menuList=new MenuList();
        for(MenuItemDTO menuItemDTO:dtos){
            if(menuItemDTO.getId()==id)
                menuList.setEntity(menuItemDTO);
        }
        List<MenuList> childs=new ArrayList<MenuList>();
        for(MenuItemDTO menuItemDTO:dtos){
            if(menuItemDTO.getParent_menu_id()!=null&&menuItemDTO.getParent_menu_id()==id){
                MenuList child=createMenu(menuItemDTO.getId(),dtos);
                childs.add(child);
            }
        }
        if(childs.size()>0)
            menuList.setChilds(childs);
        else
            menuList.setChilds(null);
        return menuList;
    }
}
